Add mean dispatch for pymc distributions #7530

lucianopaz · 2024-10-09T11:37:46Z

Description

This PR provides a single dispatch protocol that computes the expected value of most pymc distributions.

Related Issue

Closes #
Related to #

Checklist

Checked that the pre-commit linting/style checks pass
Included tests that prove the fix is effective or that the new feature works
Added necessary documentation (docstrings and/or example notebooks)
If you are a pro: each commit corresponds to a relevant logical change

Type of change

📚 Documentation preview 📚: https://pymc--7530.org.readthedocs.build/en/7530/

lucianopaz · 2024-10-09T11:38:23Z

@ricardoV94, I finally decided to put some time and port this stuff into pymc

codecov · 2024-10-09T12:11:44Z

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 92.81%. Comparing base (938aff4) to head (1c71deb).
Report is 75 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #7530      +/-   ##
==========================================
+ Coverage   92.69%   92.81%   +0.11%     
==========================================
  Files         104      106       +2     
  Lines       17409    17644     +235     
==========================================
+ Hits        16138    16376     +238     
+ Misses       1271     1268       -3

Files with missing lines	Coverage Δ
pymc/distributions/moments/__init__.py	`100.00% <100.00%> (ø)`
pymc/distributions/moments/means.py	`100.00% <100.00%> (ø)`
pymc/exceptions.py	`100.00% <100.00%> (ø)`

... and 1 file with indirect coverage changes

pymc/distributions/moments/means.py

pymc/distributions/moments/utils.py

tests/distributions/moments/test_means.py

pymc/distributions/moments/means.py

ricardoV94

Looks great. Small change requests commented above

pymc/distributions/moments/means.py

lucianopaz added the enhancements label Oct 9, 2024

lucianopaz requested a review from ricardoV94 October 9, 2024 11:37